#!/usr/bin/python

# The ChartDirector for Python module is assumed to be in "../lib"
import sys, os
sys.path.insert(0, os.path.join(os.path.abspath(sys.path[0]), "..", "lib"))

from pychartdir import *

# The XYZ points for the bubble chart
dataX0 = [170, 300, 1000, 1700]
dataY0 = [16, 69, 16, 75]
dataZ0 = [52, 105, 88, 140]

dataX1 = [500, 1000, 1300]
dataY1 = [40, 58, 85]
dataZ1 = [140, 202, 84]

# Create a XYChart object of size 540 x 480 pixels
c = XYChart(540, 480)

# Set the plotarea at (70, 65) and of size 400 x 350 pixels. Turn on both horizontal and vertical
# grid lines with light grey color (0xc0c0c0)
c.setPlotArea(70, 65, 400, 350, -1, -1, Transparent, 0xc0c0c0, -1)

# Add a legend box at (70, 30) (top of the chart) with horizontal layout. Use 12pt Times Bold Italic
# font. Set the background and border color to Transparent.
c.addLegend(70, 30, 0, "Times New Roman Bold Italic", 12).setBackground(Transparent)

# Add a title to the chart using 18pt Times Bold Itatic font.
c.addTitle("Product Comparison Chart", "Times New Roman Bold Italic", 18)

# Add titles to the axes using 12pt Arial Bold Italic font
c.yAxis().setTitle("Capacity (tons)", "Arial Bold Italic", 12)
c.xAxis().setTitle("Range (miles)", "Arial Bold Italic", 12)

# Set the axes line width to 3 pixels
c.xAxis().setWidth(3)
c.yAxis().setWidth(3)

# Add (dataX0, dataY0) as a scatter layer with red (ff3333) glass spheres, where the sphere size is
# modulated by dataZ0. This creates a bubble effect.
c.addScatterLayer(dataX0, dataY0, "Technology AAA", GlassSphere2Shape, 15, 0xff3333).setSymbolScale(
    dataZ0)

# Add (dataX1, dataY1) as a scatter layer with blue (0000ff) glass spheres, where the sphere size is
# modulated by dataZ1. This creates a bubble effect.
c.addScatterLayer(dataX1, dataY1, "Technology BBB", GlassSphere2Shape, 15, 0x0000ff).setSymbolScale(
    dataZ1)

# Output the chart
c.makeChart("threedbubble2.png")

